# encoding=utf-8
"""
@author: xiao nian
@contact: xiaonian030@163.com
@time: 2021-12-14 15:15
"""

# http 配置
HTTP_CONFIG = {
    'host': '0.0.0.0',
    'port': 55001,
    'workers': 2
}

TRAIN_CONFIG = {
    'sequence_length': 100,  # 序列长度，建议保留默认值
    'batch_size': 128,  # 批次大小，建议保留默认值
    'epochs': 100  # 训练轮次，更具实际情况调整
}

MODEL_CONFIG = {
    'name': 'muti_label_model',
    'version': '1',
    'multi_label': True,
    'directory': 'tf_serving_model',
    'embedding_type': '',  # bert
    'embedding_path': 'data/bert_model',
    'host': 'http://localhost:8501'
}
